﻿// 监控端.cpp : 定义应用程序的入口点。
//



#include "pch.h"
#include "framework.h"
#include "监控端.h"
#include <stdio.h>

#include <winsock2.h>
#pragma comment(lib,"ws2_32.lib")

//前后两张图片的发送间隔   ms
#define TIME_SPACE 50

//服务器ip地址：
#define SERVER_ADDR  "10.133.98.104"
//服务器端口号：
#define  SERVER_PORT 9527

//图片文件名：
#define BMP_FILE_NAME "caption.bmp"
//图形文件大小：
#define BMP_FILE_SIZE 6220868


#define MAX_LOADSTRING 100

// 全局变量:
HINSTANCE hInst;                                // 当前实例
WCHAR szTitle[MAX_LOADSTRING];                  // 标题栏文本
WCHAR szWindowClass[MAX_LOADSTRING];            // 主窗口类名

//服务器和客户端的socket：
SOCKET g_sSocket, g_cSocket;



// 此代码模块中包含的函数的前向声明:
ATOM                MyRegisterClass(HINSTANCE hInstance);
BOOL                InitInstance(HINSTANCE, int);
LRESULT CALLBACK    WndProc(HWND, UINT, WPARAM, LPARAM);
INT_PTR CALLBACK    About(HWND, UINT, WPARAM, LPARAM);



void OnCreate(HWND hWnd);
//接收文件并显示：
void recvAndShow(HWND hWNd);




int APIENTRY wWinMain(_In_ HINSTANCE hInstance,
                     _In_opt_ HINSTANCE hPrevInstance,
                     _In_ LPWSTR    lpCmdLine,
                     _In_ int       nCmdShow)
{
    UNREFERENCED_PARAMETER(hPrevInstance);
    UNREFERENCED_PARAMETER(lpCmdLine);

    // TODO: 在此处放置代码。

    // 初始化全局字符串
    LoadStringW(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
    LoadStringW(hInstance, IDC_MY, szWindowClass, MAX_LOADSTRING);
    MyRegisterClass(hInstance);

    // 执行应用程序初始化:
    if (!InitInstance (hInstance, nCmdShow))
    {
        return FALSE;
    }

    HACCEL hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_MY));

    MSG msg;

    // 主消息循环:
    while (GetMessage(&msg, nullptr, 0, 0))
    {
        if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
        {
            TranslateMessage(&msg);
            DispatchMessage(&msg);
        }
    }

    return (int) msg.wParam;
}



//
//  函数: MyRegisterClass()
//
//  目标: 注册窗口类。
//
ATOM MyRegisterClass(HINSTANCE hInstance)
{
    WNDCLASSEXW wcex;

    wcex.cbSize = sizeof(WNDCLASSEX);

    wcex.style          = CS_HREDRAW | CS_VREDRAW;
    wcex.lpfnWndProc    = WndProc;
    wcex.cbClsExtra     = 0;
    wcex.cbWndExtra     = 0;
    wcex.hInstance      = hInstance;
    wcex.hIcon          = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_MY));
    wcex.hCursor        = LoadCursor(nullptr, IDC_ARROW);
    wcex.hbrBackground  = (HBRUSH)(COLOR_WINDOW+1);
    wcex.lpszMenuName   = MAKEINTRESOURCEW(IDC_MY);
    wcex.lpszClassName  = szWindowClass;
    wcex.hIconSm        = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));

    return RegisterClassExW(&wcex);
}

//
//   函数: InitInstance(HINSTANCE, int)
//
//   目标: 保存实例句柄并创建主窗口
//
//   注释:
//
//        在此函数中，我们在全局变量中保存实例句柄并
//        创建和显示主程序窗口。
//
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
{
   hInst = hInstance; // 将实例句柄存储在全局变量中

   HWND hWnd = CreateWindowW(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
      CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, nullptr, nullptr, hInstance, nullptr);

   if (!hWnd)
   {
      return FALSE;
   }

   ShowWindow(hWnd, nCmdShow);
   UpdateWindow(hWnd);

   return TRUE;
}

//
//  函数: WndProc(HWND, UINT, WPARAM, LPARAM)
//
//  目标: 处理主窗口的消息。
//
//  WM_COMMAND  - 处理应用程序菜单
//  WM_PAINT    - 绘制主窗口
//  WM_DESTROY  - 发送退出消息并返回
//
//
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
    switch (message)
    {
    case WM_TIMER:
        recvAndShow(hWnd);
        break;
    case WM_CREATE:
        OnCreate(hWnd);
        break;
    case WM_COMMAND:
        {
            int wmId = LOWORD(wParam);
            // 分析菜单选择:
            switch (wmId)
            {
            case IDM_ABOUT:
                DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
                break;
            case IDM_EXIT:
                DestroyWindow(hWnd);
                break;
            default:
                return DefWindowProc(hWnd, message, wParam, lParam);
            }
        }
        break;
    case WM_PAINT:
        {
            PAINTSTRUCT ps;
            HDC hdc = BeginPaint(hWnd, &ps);
            // TODO: 在此处添加使用 hdc 的任何绘图代码...



            HDC hShowDc = CreateCompatibleDC(hdc);
            //连通上网络之后 发送过来：
            HANDLE hBitmap = LoadImage(hInst, BMP_FILE_NAME, IMAGE_BITMAP, 1920, 1080, LR_LOADFROMFILE);
            SelectObject(hShowDc, hBitmap);
            BitBlt(hdc, 0, 0, 1920, 1080, hShowDc, 0, 0, SRCCOPY);




            EndPaint(hWnd, &ps);
        }
        break;
    case WM_DESTROY:
        KillTimer(hWnd, 10086);
        closesocket(g_sSocket);
        closesocket(g_cSocket);
        PostQuitMessage(0);
        break;
    default:
        return DefWindowProc(hWnd, message, wParam, lParam);
    }
    return 0;
}

// “关于”框的消息处理程序。
INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
    UNREFERENCED_PARAMETER(lParam);
    switch (message)
    {
    case WM_INITDIALOG:
        return (INT_PTR)TRUE;

    case WM_COMMAND:
        if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
        {
            EndDialog(hDlg, LOWORD(wParam));
            return (INT_PTR)TRUE;
        }
        break;
    }
    return (INT_PTR)FALSE;
}

//接收文件并显示：
void recvAndShow(HWND hWnd)
{
    //打开文件：
    FILE* fp = fopen(BMP_FILE_NAME, "wb");

    //文件大小：
    int fileSize = BMP_FILE_SIZE;

    //发送文件内容：
    char temp[4096];
    int count = 0;
    int r;
    while (1)
    {
        r = recv(g_cSocket, temp, 4096, NULL);

        if (r > 0)
        {
            fwrite(temp, 1, r, fp);
            count += r;
            if (count >= fileSize)
            {
                break;
            }
        }
        
    }


    //关闭文件：
    fclose(fp);

    //显示到窗口上：
    HDC hdc = GetDC(hWnd);

    HDC hShowDc = CreateCompatibleDC(hdc);
    //连通上网络之后   发送过来：
    HANDLE hBitmap = LoadImage(hInst, BMP_FILE_NAME, IMAGE_BITMAP, 1920, 1080, LR_LOADFROMFILE);

    SelectObject(hShowDc, hBitmap);

    BitBlt(hdc, 0, 0, 1920, 1080, hShowDc, 0, 0, SRCCOPY);

    ReleaseDC(hWnd, hdc);

}

void OnCreate(HWND hWnd)
{
    //tcp网络连接：
    WSADATA wsaData;
    WSAStartup(MAKEWORD(2, 2), &wsaData);
  
    g_sSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    SOCKADDR_IN sAddr = { 0 };
    sAddr.sin_family = AF_INET;
    sAddr.sin_addr.S_un.S_addr = inet_addr(SERVER_ADDR);
    sAddr.sin_port = htons(SERVER_PORT);

    bind(g_sSocket, (sockaddr*)&sAddr, sizeof(sAddr));

    listen(g_sSocket, 10);

    SOCKADDR_IN cAddr = { 0 };
    int len = sizeof(cAddr);

    g_cSocket = accept(g_sSocket, (sockaddr*)&cAddr, &len);

    //recvAndShow(hWnd);

    SetTimer(hWnd, 10086, TIME_SPACE, NULL);

}